# states/__init__.py（状态管理模块导出）
from .global_state import GlobalState
from .agent_state import AgentState
from .task_state import TaskStateManager
from .langgraph_workflow import LangGraphWorkflowEngine as LangGraphWorkflow
from .context_partition_manager import (
    ContextPartitionManager, 
    ContextPartitionType, 
    PartitionStrategy,
    ContextEntry,
    PartitionConfig,
    context_partition_manager
)

__all__ = [
    "GlobalState",
    "AgentState",
    "TaskStateManager",
    "LangGraphWorkflow",  # 作为LangGraphWorkflowEngine的别名导出
    "ContextPartitionManager",
    "ContextPartitionType",
    "PartitionStrategy",
    "ContextEntry",
    "PartitionConfig",
    "context_partition_manager"
]